from .nnmodel import NNModel, UniMolModel